import numpy as np

from core.variable import Variable
from utils.functions_collect import matmul

c0 = Variable(np.array([[1, 0, 0, 0, 0, 0, 0]]))
c1 = Variable(np.array([[0, 0, 1, 0, 0, 0, 0]]))
W_in = np.random.randn(7, 3)
W_out = np.random.randn(3, 7)
h0 = matmul(c0, W_in)
h1 = matmul(c1, W_in)
h = h0 + h1
s = matmul(h, W_out)
print(s)
